import torch
from A5_implement_multilayer_neural_networks import NeuralNetwork


model = NeuralNetwork(num_inputs=2, num_outputs=2)
model.load_state_dict(torch.load("model.pth"))